"""量化模型模块"""
from .base_model import BaseQuantModel
from .feature_engineering import (
    create_features,
    create_target,
    calculate_sma,
    calculate_ema,
    calculate_macd,
    calculate_rsi,
    calculate_atr,
    calculate_bollinger_bands,
    calculate_momentum,
    calculate_volume_features,
    calculate_volatility_features
)
from .logistic_regression import LogisticRegressionModel
from .linear_regression import LinearRegressionModel
from .xgboost_model import XGBoostModel
from .lightgbm_model import LightGBMModel
from .lstm_model import LSTMModel
from .gru_model import GRUModel
from .temporal_cnn_model import TemporalCNNModel

__all__ = [
    # 基础类
    'BaseQuantModel',
    # 特征工程
    'create_features',
    'create_target',
    'calculate_sma',
    'calculate_ema',
    'calculate_macd',
    'calculate_rsi',
    'calculate_atr',
    'calculate_bollinger_bands',
    'calculate_momentum',
    'calculate_volume_features',
    'calculate_volatility_features',
    # 模型类
    'LogisticRegressionModel',
    'LinearRegressionModel',
    'XGBoostModel',
    'LightGBMModel',
    'LSTMModel',
    'GRUModel',
    'TemporalCNNModel',
]

